package net.kelissa.divizor.model;

import java.util.HashMap;
import java.util.Map;

public class ClassWeight
{
  private Map<String, Double> weights;

  public ClassWeight()
  {
    this.weights = new HashMap<String, Double>();
  }

  public double getWeigth( String clazz)
  {
    double res = weights.containsKey(clazz) ? weights.get(clazz) : 1;
    return res;
  }

  public void putWeigth( String clazz, double w)
  {
   weights.put(clazz, w);
  }

  public Map<String, Double> getWeights()
  {
    return weights;
  }

  public void setWeights(Map<String, Double> weights)
  {
    this.weights = weights;
  }
}
